package 层序遍历;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

/**
 * @Date 2024/9/8 22:17
 * @description: .N 叉树的层序遍历以及锯齿遍历
 * @Author LittleNight
 */
public class Node {

    public int val;
    public List<Node> children;

    public Node() {}

    public Node(int _val) {
        val = _val;
    }

    public Node(int _val, List<Node> _children) {
        val = _val;
        children = _children;
    }

    // 队列用于 深搜, 这个是基础一定要会. 很多题都是这个思路
    public List<List<Integer>> levelOrder(Node root) {
        List<List<Integer>> ret = new ArrayList<>();
        if(root == null) return new ArrayList<>();
        // 创建一个队列. 双向链表做队列
        Queue<Node> queue = new LinkedList<>();
        queue.offer(root);
        
        // 层序遍历主逻辑
        while(!queue.isEmpty()) {
            int sz = queue.size();
            List<Integer> tmp = new ArrayList<>();
            for (int i = 0; i < sz; i++) {
                // 弹出队列元素, 加入到结果中, 并把孩子节点加入到队列中
                Node node = queue.poll();
                tmp.add(node.val);
                for (Node child : node.children) {
                    if(child != null) queue.offer(child);
                }
            }
            ret.add(tmp);
        }
        return ret;
    }
}
